package com.tanhua.sso.service;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.dubbo.config.annotation.Reference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.common.mapper.UserMapper;
import com.tanhua.common.pojo.User;
import com.tanhua.dubbo.server.api.HuanXinApi;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class UserService {
    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")// 盐值
    private String secret;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Reference(version = "1.0.0")
    private HuanXinApi huanXinApi;

    /**
     * 用户登录
     *
     * @param phone            用户手机号
     * @param verificationCode 验证码
     * @return token + "|" + isNew
     */
    public Map<String, String> login(String phone, String verificationCode) {
        String redisKey = "CHECK_CODE_" + phone;
        boolean isNew = false;

        // 去Redis中查找有无对应的验证码
        String redisData = redisTemplate.opsForValue().get(redisKey);
        if (!StringUtils.equals(verificationCode, redisData)) {
            return null;//验证码不存在或者不对应,则返回null
        }

        // 走到这说明验证码验证成功,验证码校验完成,将验证码废弃掉
        redisTemplate.delete(redisKey);

        // 去数据库中查询有无此手机号,即确认这手机号是不是已注册用户
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", phone);
        User user = userMapper.selectOne(queryWrapper);

        // 如手机号未注册,则注册
        if (null == user) {
            // 注册用户
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));//默认初始密码
            userMapper.insert(user);//数据库自增长的id会被自动注入对象

            isNew = true;

            // 将该用户信息注册到环信平台
            Boolean registerResult = huanXinApi.register(user.getId());
            if (!registerResult) {
                log.error("注册到环信平台失败! userId = " + user.getId());
            }
        }

        // 走到这一步说明改手机号已完成注册
        // 生成token
        Map<String, Object> claims = new HashMap<>();
        claims.put("id", user.getId());
        String token = Jwts.builder()
                .setClaims(claims)
                .signWith(SignatureAlgorithm.HS256, secret)
                .setExpiration(new DateTime().plusYears(100).toDate())//设置12h后过期
                .compact();

        // mq
        try {
            // 发送用户登录成功的消息
            HashMap<String, Object> msg = new HashMap<>();
            msg.put("id", user.getId());
            msg.put("date", System.currentTimeMillis());

            rocketMQTemplate.convertAndSend("tanhua-sso-login", msg);
        } catch (MessagingException e) {
            log.error("发送消息失败", e);
        }

        Map<String, String> tokenAndIsNew = new HashMap<>();
        tokenAndIsNew.put("token", token);
        tokenAndIsNew.put("isNew", String.valueOf(isNew));
        return tokenAndIsNew;
    }

    public User queryUserByToken(String token) {
        try {
            // 通过token解析数据
            Map<String, Object> body = Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            // sample : body = {mobile=1333333333, id=2, exp=1605513392}
            User user = new User();
            user.setId(Long.valueOf(body.get("id").toString()));

            // 需要返回user对象中的mobile,需要查询数据库获取到mobile
            // 如果每次都查询数据库,必然会导致性能问题,所以需要对用户手机号进行缓存操作
            // 数据缓存时,需要设置过期时间,过期时间要和token的时间一致
            // 如果用户修改了手机号,需要同步修改redis中的数据

            // 先去redis中查查看有没有
            String redisKey = "TANHUA_USER_MOBILE_" + user.getId();
            if (Boolean.TRUE == redisTemplate.hasKey(redisKey)) {
                String mobile = redisTemplate.opsForValue().get(redisKey);
                user.setMobile(mobile);
            } else {
                // redis中没有,我们来查数据库有没有
                // 查询数据库
                User byId = userMapper.selectById(user.getId());
                user.setMobile(byId.getMobile());

                // 将手机号写入到redis
                // 在JWT中过期时间单位为s
                long timeout = Long.parseLong(body.get("exp").toString()) * 1000 - System.currentTimeMillis();
                redisTemplate.opsForValue().set(redisKey, user.getMobile(), timeout, TimeUnit.MILLISECONDS);
            }
            return user;
        } catch (ExpiredJwtException e) {
            log.info("token已经过期! token = " + token);
            System.out.println("token已经过期！");
        } catch (Exception e) {
            log.error("token不合法! token = " + token, e);
        }
        return null;
    }

    public Boolean updatePhone(Long userId, String newPhone) {
        //先查询新手机号是否已经注册，如果已经注册，就不能修改
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", newPhone);
        User user = userMapper.selectOne(queryWrapper);
        if (ObjectUtil.isNotEmpty(user)) {
            //新手机号已经被注册
            return false;
        }

        user = new User();
        user.setId(userId);
        user.setMobile(newPhone);

        return userMapper.updateById(user) > 0;
    }
}
